import torch
import torch.nn.functional as F
from mmcv.cnn import ConvModule, kaiming_init, build_conv_layer
from mmcv.runner import force_fp32
from torch import nn
from .transfusion_head import TransFusionHead
from mmdet3d.models.builder import HEADS
from .. import builder
from ..builder import HEADS, build_loss

@HEADS.register_module()
class DummyHead( ):
    def __init__(self, num_classes=0, hidden_channel=128, **kwargs):
        super(DummyHead, self).__init__()
        self.num_classes = num_classes
        self.hidden_channel = hidden_channel
    def forward(self, x):
        return x
    def loss(self, *args, **kwargs):
        return {}
    def get_bboxes(self, *args, **kwargs):
        return []